home *** CD-ROM | disk | FTP | other *** search
/ C/C++ Users Group Library 1997 August / Walnut Creek CDROM.7z / VOL_400 / 466_01 / SRC / ADOUTPUT.CPP < prev    next >
Encoding:
C/C++ Source or Header  |  1996-12-20  |  1.9 KB  |  91 lines
  1. #include <afx.h>
  2. #include "adoutput.h"
  3. #include "errmsg.h"
  4.  
  5. //@doc FILE ADOUTPUT
  6.  
  7.  
  8. //@mfunc Constructor. Initializes the output buffer & count vars.
  9. CAdOutput::CAdOutput()
  10. { 
  11.     m_szOutBuf = NULL;
  12.     m_nBufLen = 0;
  13.     m_nBufCur = 0;
  14. }
  15.  
  16. //@mfunc Destructor. Deletes the output buffer if present.
  17.  
  18. CAdOutput::~CAdOutput()
  19. {
  20.     if(m_szOutBuf)
  21.         delete m_szOutBuf;
  22. }
  23.  
  24. //@mfunc Allocates a memory block for the output buffer and resets
  25. // count variables. 
  26. //@rdesc Zero if successful or errMemory if not enough memory for requested
  27. // buffer.
  28. int CAdOutput::SetOutputBuffer(
  29.     UINT nBufLen)           //@parm How much memory to allocate.
  30. {
  31. TRY
  32. {
  33.     if(m_szOutBuf)
  34.         delete m_szOutBuf;
  35.  
  36.     m_szOutBuf = new char[nBufLen];
  37.     m_nBufLen = nBufLen;
  38.     m_nBufCur = 0;
  39.  
  40.     return 0;
  41. }
  42. CATCH(CMemoryException, e)
  43. {
  44.     return errMemory;
  45. }
  46. END_CATCH
  47. }
  48.  
  49.  
  50. //@mfunc Writes all unwritten text to the output file.
  51.  
  52. void CAdOutput::FlushOutputBuffer(void)
  53. {
  54.     ASSERT(m_szOutBuf);
  55.  
  56.     if(m_nBufCur)
  57.     {
  58.         CFile::Write(m_szOutBuf, m_nBufCur);
  59.         m_nBufCur = 0;
  60.     }
  61. }
  62.  
  63. //@mfunc Writes a chunk of data to the output buffer, flushing the
  64. // buffer first if necessary. Note: Do NOT attempt to write a data
  65. // block larger than the allocated buffer size. If you do so the
  66. // write will be truncated, only the data equal to the buffer size
  67. // will be written, and no error message will occur.
  68.  
  69. void CAdOutput::Write(
  70.     const void* lpBuf,      //@parm Pointer to stuff to write
  71.     UINT nCount)            //@parm How many bytes to write
  72. {
  73.     ASSERT(nCount <= m_nBufLen);
  74.     ASSERT(m_szOutBuf);
  75.  
  76.     if(nCount + m_nBufCur <= m_nBufLen)
  77.     {
  78.         memcpy(m_szOutBuf+m_nBufCur, lpBuf, nCount);
  79.         m_nBufCur += nCount;
  80.     }
  81.     else
  82.     {
  83.         FlushOutputBuffer();
  84.  
  85.         if(nCount > m_nBufLen)
  86.             nCount = m_nBufLen;
  87.  
  88.         Write(lpBuf, nCount);
  89.     }
  90. }
  91.